import tkinter as tk
import os
from tkinter import ttk
from login import *
from outgoing import *
from incoming import *

def check_make_dir():
    '''
    检查所需文件夹是否存在, 若不存在则创建
    '''
    if not os.path.exists("./入库"):
        os.mkdir("./入库")
    if not os.path.exists("./出库"):
        os.mkdir("./出库")
    if not os.path.exists("./汇总"):
        os.mkdir("./汇总")
    if not os.path.exists("./汇总/日汇总"):
        os.mkdir("./汇总/日汇总")
    if not os.path.exists("./汇总/月汇总"):
        os.mkdir("./汇总/月汇总")
    if not os.path.exists("./汇总/年汇总"):
        os.mkdir("./汇总/年汇总")

def main(user):
    root = tk.Tk()
    root.title("仓库管理系统")
    width = 800
    height = 320
    screenwidth = root.winfo_screenwidth()
    screenheight = root.winfo_screenheight()
    root.geometry('%dx%d+%d+%d'%(width, height, (screenwidth-width)/2, (screenheight-height)/2))


    tab_control = ttk.Notebook(root)
    in_tab = ttk.Frame(tab_control)
    incoming(root, in_tab)
    tab_control.add(in_tab, text='入库管理')
    

    out_tab = ttk.Frame(tab_control)
    outgoing(root, out_tab)
    tab_control.add(out_tab, text='出库管理')
    
    
    if user == "仓库管理员":
        from summary import summary
        summary_tab = ttk.Frame(tab_control)
        summary(summary_tab)
        tab_control.add(summary_tab, text='汇总管理')

    tab_control.pack(expand=1, fill="both")

    root.mainloop()
if __name__ == "__main__":
    check_make_dir()
    login()